package com.lbc.hashtab;

import java.util.Scanner;

public class HashTabDemo {
    // 用数组和单链表实现一个哈希表
    public static void main(String[] args) {
        // 创建哈希表
        HashTab hashTab = new HashTab(7);

        //  写一个菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add：添加雇员");
            System.out.println("list：显示雇员");
            System.out.println("find：查找雇员");
            System.out.println("exit：退出系统");

            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    // 创建雇员
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("输入id");
                    id = scanner.nextInt();
                    hashTab.findEmpById(id);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                    break;
                default:
                    break;
            }
        }
    }

}

// 创建HashTab 管理多条链表
class HashTab {
    private EmpLinkedList[] empLinkedListArrays;
    private int size;

    // 构造器
    public HashTab (int size) {
        this.size = size;
        // 初始化
        this.empLinkedListArrays = new EmpLinkedList[size];
        // 还要对数组中的每一个 EmpLinkedList 初始化
        for (int i = 0; i < size; i++) {
            empLinkedListArrays[i] = new EmpLinkedList();
        }
    }

    // 添加雇员
    public void add(Emp emp) {
        // 根据员工的id，得到要添加的链表
        int empLinkedListNo = hashFun(emp.id);
        // 将emp 添加到相应的链表中
        empLinkedListArrays[empLinkedListNo].add(emp);
    }
    // 遍历所有的链表，遍历hashtab
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedListArrays[i].list(i);
        }
    }
    // 根据id查找雇员
    public void findEmpById (int id) {
        // 使用散列函数确定到哪条链表查找
        int empLinkedListNo = hashFun(id);
        Emp emp = empLinkedListArrays[empLinkedListNo].findEmpById(id);
        if (emp != null) {  // 找到
            System.out.printf("在%d条链表中找到雇员 id=%d name=%s\n", empLinkedListNo+1, id, emp.name);
        } else {
            System.out.println("没有找到该雇员");
        }
    }

    // 编写散列函数，使用一个简单取模法
    public int hashFun (int id) {
        return id % size;
    }

}

// 表示一个雇员
class Emp {
    public int id;
    public String name;
    public Emp next;  //next 默认为 null

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

// 创建 EmpLinkedList , 表示链表
class EmpLinkedList {
    // 头指针，指向第一个雇员，因此我们这个链表的head是直接指向第一个雇员的
    private Emp head;  // 默认为 null

    // 添加雇员到链表
    // 说明
    // 1.假定，当添加雇员时，id 是自增的， 即id的分配总是从小到大
    //   因此我们将该雇员直接加入到本链表的最后即可
    public void add(Emp emp) {
        // 如果是添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        // 如果不是第一个雇员，则使用一个辅助的指针，帮助定位到最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.next == null) {  // 说明到链表最后
                break;
            }
            curEmp = curEmp.next;  // 后移
        }
        // 退出时直接将 emp 加入链表
        curEmp.next = emp;
    }

    // 遍历链表的雇员信息
    public void list(int no) {
        if (head == null) {
            System.out.println("第 "+(no+1)+" 链表为空");
            return;
        }
        System.out.print("第 "+(no+1)+" 链表链表的信息为：");
        Emp curEmp = head;  // 辅助指针
        while (true) {
            System.out.printf("=> id=%d name=%s\t", curEmp.id, curEmp.name);
            if (curEmp.next == null) {  // 说明 curEmp 已经是最后一个元素了
                break;
            }
            curEmp = curEmp.next;
        }
        System.out.println();
    }

    // 根据id查找雇员
    public Emp findEmpById (int id) {
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        Emp curEmp = head;
        while (true) {
            if (curEmp.id == id) {
                break;  // 这是curEmp 就指向要查找的雇员
            }
            if (curEmp.next == null) {  // 说明遍历当前链表没有找到该雇员
                curEmp = null;
                break;
            }
            curEmp = curEmp.next;
        }
        return curEmp;
    }

    // 修改
    // 删除
}
